package com.yhsimon.seckill.utils;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.yhsimon.seckill.pojo.User;
import com.yhsimon.seckill.vo.RespBean;

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.Timestamp;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * @author： YHSimon
 * @date： 2022-04-25 13:46
 */
public class UserUtil {
    private static void createUser(int count) throws Exception {
        System.out.println("start create user");
        List<User> users = new ArrayList<>(count);
        for(int i=0;i<count;++i){
            User user = new User();
            user.setId(13000000000L+i);
            user.setNickname("user"+i);
            user.setSalt("1a2b3c4d");
            user.setPassword(MD5Util.inputPassToDBPass("123456", user.getSalt()));
            user.setLoginCount(1);
            user.setRegisterDate(new Date());
            users.add(user);
        }
        System.out.println("finish create user");
        // Connection conn = getConn();
        // String sql="insert into t_user(login_count,nickname,register_date,salt,password,id) values(?,?,?,?,?,?)";
        // PreparedStatement pst = conn.prepareStatement(sql);
        // for(User user:users){
        //     pst.setInt(1, user.getLoginCount());
        //     pst.setString(2, user.getNickname());
        //     pst.setTimestamp(3, new Timestamp(user.getRegisterDate().getTime()));
        //     pst.setString(4, user.getSalt());
        //     pst.setString(5, user.getPassword());
        //     pst.setLong(6, user.getId());
        //     pst.addBatch();
        // }
        // pst.executeBatch();
        // pst.clearParameters();
        // conn.close();
        // System.out.println("insert into db");

        // 登录，生成userTicket
        String urlString="http://localhost:8080/login/doLogin";
        File file = new File("C:/Users//Q倘若/Desktop/config.txt");
        if(file.exists()){
            file.delete();
        }
        RandomAccessFile raf = new RandomAccessFile(file, "rw");
        file.createNewFile();
        raf.seek(0);
        for (User user:users){
            URL url = new URL(urlString);
            HttpURLConnection co = (HttpURLConnection) url.openConnection();
            co.setRequestMethod("POST");
            co.setDoOutput(true);
            OutputStream out = co.getOutputStream();
            String params="mobile="+user.getId()+"&password="+MD5Util.inputPassToFromPass("123456");
            out.write(params.getBytes());
            out.flush();
            InputStream inputStream = co.getInputStream();
            ByteArrayOutputStream bout = new ByteArrayOutputStream();
            byte[] buff = new byte[1024];
            int len=0;
            while((len=inputStream.read(buff))>=0){
                bout.write(buff, 0, len);
            }
            inputStream.close();
            bout.close();
            String response = new String(bout.toByteArray());
            ObjectMapper mapper = new ObjectMapper();
            RespBean respBean = mapper.readValue(response, RespBean.class);
            String userTicket = (String) respBean.getObj();
            System.out.println("create userTicket："+user.getId());
            String row = user.getId() + "," + userTicket;
            raf.seek(raf.length());
            raf.write(row.getBytes());
            raf.write("\r\n".getBytes());
            System.out.println("write to file: "+user.getId());

        }
        raf.close();
        System.out.println("over");

    }

    private static Connection getConn() throws Exception {
        String url="jdbc:mysql://localhost:3306/seckill?userUnicode=true&characterEncoding=UTF-8&serverTimezone=Asia/Shanghai";
        String username="root";
        String password="123456";
        String driver="com.mysql.cj.jdbc.Driver";
        Class.forName(driver);
        return DriverManager.getConnection(url, username, password);
    }

    public static void main(String[] args) throws Exception {
        createUser(5000);
    }
}
